package net.longjin.feignClient;

import feign.RequestTemplate;
import feign.codec.EncodeException;
import feign.codec.Encoder;
import feign.form.ContentType;
import feign.form.FormEncoder;
import feign.form.MultipartFormContentProcessor;
import feign.form.spring.SpringManyMultipartFilesWriter;
import feign.form.spring.SpringSingleMultipartFileWriter;
import org.springframework.beans.factory.ObjectFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.http.HttpMessageConverters;
import org.springframework.cloud.openfeign.support.SpringEncoder;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.multipart.MultipartFile;

import java.lang.reflect.Type;
import java.util.*;

/***
 * @author: 何志鹏
 * @Date: 2019/10/10
 */
@Configuration
public class FeignMultipartSupportConfig {

    @Autowired
    private ObjectFactory<HttpMessageConverters> messageConverters;

    @Bean
    public Encoder feignEncoder() {
        return new SpringMultipartEncoder(new SpringEncoder(messageConverters));
    }

    private class SpringMultipartEncoder extends FormEncoder {
        public SpringMultipartEncoder(Encoder delegate) {
            super(delegate);
            MultipartFormContentProcessor processor = (MultipartFormContentProcessor) this.getContentProcessor(ContentType.MULTIPART);
            processor.addFirstWriter(new SpringSingleMultipartFileWriter());
            processor.addFirstWriter(new SpringManyMultipartFilesWriter());
        }

        @Override
        public void encode(Object object, Type bodyType, RequestTemplate template) throws EncodeException {
            if (bodyType.equals(MultipartFile[].class)) {
                MultipartFile[] files = (MultipartFile[]) ((MultipartFile[]) object);
                HashMap data = new HashMap(files.length, 1.0F);
                data.put(files[0].getName(), Arrays.asList(files));
                super.encode(data, MAP_STRING_WILDCARD, template);
            } else if (bodyType.equals(MultipartFile.class)) {
                MultipartFile file = (MultipartFile) object;
                Map<String, Object> data = Collections.singletonMap(file.getName(), object);
                super.encode(data, MAP_STRING_WILDCARD, template);
            } else if ((object instanceof Iterable) && ((Iterable) object).iterator().hasNext() && ((Iterable) object).iterator().next() instanceof MultipartFile) {
                Iterable<?> iterable = (Iterable) object;
                HashMap data = new HashMap();
                Iterator var13 = iterable.iterator();

                while (var13.hasNext()) {
                    Object item = var13.next();
                    MultipartFile file = (MultipartFile) item;
                    data.put(file.getName(), file);
                }

                super.encode(data, MAP_STRING_WILDCARD, template);
            } else {
                super.encode(object, bodyType, template);
            }
        }
    }
}
